#ifndef __MYTREEITEMDATA_H__
#define __MYTREEITEMDATA_H__

#include "wx/treectrl.h"
#include "../src/scene.h"

class MyTreeItemData : public wxTreeItemData{
    public:
        MyTreeItemData(Solid *);
        MyTreeItemData(Face *);
        MyTreeItemData(Edge *);
        MyTreeItemData(Vertex *);
        MyTreeItemData(Point *);
        ~MyTreeItemData();

        Solid *getSolid();
        Face *getFace();
        Edge *getEdge();
        Vertex *getVertex();
        Point *getPoint();
        void select(bool, Scene *);

    private:
        Solid *s;
        Face *f;
        Edge *e;
        Vertex *v;
        Point *p;
};

#endif
